import cv2
import matplotlib.pyplot as plt

img = cv2.imread("assets/p1.jpg")
img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
h, s, v = cv2.split(img)

plt.figure("Original VS Equalized")
plt.subplot(1, 3, 1)
plt.hist(h.ravel(), 256)
plt.subplot(132)
plt.hist(s.ravel(), 256)
plt.subplot(133)
plt.hist(v.ravel(), 256)
plt.show()

#equ_b = cv2.equalizeHist(h)
#equ_g = cv2.equalizeHist(s)
equ_v = cv2.equalizeHist(v)

equ_img = cv2.merge((h, s, equ_v))
cv2.imshow("original",cv2.cvtColor(img, cv2.COLOR_HSV2BGR))
cv2.imshow("equalize",cv2.cvtColor(equ_img, cv2.COLOR_HSV2BGR))
cv2.waitKey(0)
